Trace lithium dose plus treatment as usual